"""
Pydantic 数据模式包
定义API输入输出的数据结构
"""

from .user import (
    UserBase, UserCreate, UserUpdate, UserInDB, User,
    UserLogin, UserRegister, Token, TokenData
)
from .project import (
    ProjectBase, ProjectCreate, ProjectUpdate, ProjectInDB, Project,
    ProjectSummary, ProjectList
)
from .data_source import (
    DataSourceBase, DataSourceCreate, DataSourceUpdate, DataSourceInDB, DataSource,
    DataSourceTest, DataSourceList
)
from .training_job import (
    TrainingJobBase, TrainingJobCreate, TrainingJobUpdate, TrainingJobInDB, TrainingJob,
    TrainingJobList, TrainingJobSummary
)
from .model import (
    ModelBase, ModelCreate, ModelUpdate, ModelInDB, Model,
    ModelList, ModelSummary, ModelDeploy
)
from .common import (
    BaseResponse, ListResponse, PaginatedResponse,
    StatusResponse, MessageResponse
)

__all__ = [
    # User schemas
    "UserBase", "UserCreate", "UserUpdate", "UserInDB", "User",
    "UserLogin", "UserRegister", "Token", "TokenData",
    
    # Project schemas
    "ProjectBase", "ProjectCreate", "ProjectUpdate", "ProjectInDB", "Project",
    "ProjectSummary", "ProjectList",
    
    # DataSource schemas
    "DataSourceBase", "DataSourceCreate", "DataSourceUpdate", "DataSourceInDB", "DataSource",
    "DataSourceTest", "DataSourceList",
    
    # TrainingJob schemas
    "TrainingJobBase", "TrainingJobCreate", "TrainingJobUpdate", "TrainingJobInDB", "TrainingJob",
    "TrainingJobList", "TrainingJobSummary",
    
    # Model schemas
    "ModelBase", "ModelCreate", "ModelUpdate", "ModelInDB", "Model",
    "ModelList", "ModelSummary", "ModelDeploy",
    
    # Common schemas
    "BaseResponse", "ListResponse", "PaginatedResponse",
    "StatusResponse", "MessageResponse"
]